package com.test.spring.util.aspect.datasource;

/**
 * 线程相关的数据源处理类
 * <p>
 * Created by ll on 2017/2/13.
 */
public class DataSourceHandle {

    //线程本地环境
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    /**
     * 设置数据源
     * @param dataSource 数据源名称
     */
    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

    /**
     * 获取数据源
     * @return 数据源名称
     */
    public static String getDataSource() {
        return contextHolder.get();
    }

    /**
     * 清空数据源
     */
    public static void cleanDataSource() {
        contextHolder.remove();
    }
}
